#!/usr/bin/env python 
# -*- coding:utf-8 -*-
# 作者：chenping
# 问题：SyntaxError: (unicode error) 'unicodeescape' codec can't decode bytes in position 2-3: truncated \UXXXXXXXX escape
# 解决---- \ 改为 /
from PIL import Image
import cv2
import numpy as np
import getpath 
import os
# 读取照片
fname=os.path.join(getpath.getpath(),"images","zp.jpg")
img = cv2.imread(fname)
 
# 缩小图片
img = cv2.resize(img, None, fx=0.25, fy=0.25)
rows, cols, channels = img.shape
logging.info(rows, cols, channels)
# 展示图片
cv2.imshow("img", img)
 
# 图片转换为灰度图并显示
hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
# 显示图像
# cv2.imshow('hsv', hsv)
 
# 图片的二值化处理
# 红底变蓝底
# 将在两个阈值内的像素值设置为白色（255），而不在阈值区间内的像素值设置为黑色（0）
# lower_red= np.array([0,135,135])
# upper_red = np.array([180,245,230])
# mask = cv2.inRange(hsv, lower_red, upper_red)
# # 显示图像
# cv2.imshow('mask',mask)
 
 
# 蓝底变红底
lower_blue = np.array([90, 70, 70])
upper_blue = np.array([110, 255, 255])
mask = cv2.inRange(hsv, lower_blue, upper_blue)
# 显示图像
# cv2.imshow('mask', mask)
 
# 腐蚀膨胀 若是腐蚀膨胀后仍有白色噪点，可以增加iterations的值
erode = cv2.erode(mask, None, iterations=2)
# cv2.imshow('erode', erode)
 
dilate = cv2.dilate(erode, None, iterations=2)
cv2.imshow('dilate', dilate)
 
 
# 遍历每个像素点，进行颜色的替换
# img[i,j]=(0,0,255)： 此处替换颜色，为BGR通道，不是RGB通道，若是想要将红底变成蓝底img[i,j]=(255,0,0)，
# 若是想将蓝底变为红底则img[i,j]=(0,0,255),若是想变白底img[i,j]=(255,255,255)
for i in range(rows):
    for j in range(cols):
        if dilate[i, j] == 255:  # 像素点255表示白色
            img[i, j] = (255, 255, 255)  # 此处替换颜色，为BGR通道，不是RGB通道
cv2.imwrite('blue-white.jpg', img)
 
# res = cv2.imread("C:/Users/chenping/Desktop/blue-white.jpg")
# cv2.imshow('res', res)
# # 窗口等待的命令，0表示无限等待
# cv2.waitKey(0)